from io import BytesIO
import pytesseract,requests
from PIL import Image

def get_verification_code():
    '''
    func:获取验证码。
    注意：需要安装安装 Tesseract-OCR。
    :return:
    '''
    # 设置Tesseract路径（Windows用户可能需要）
    pytesseract.pytesseract.tesseract_cmd = r"D:\Tesseract-OCR\tesseract.exe"
    # 获取验证码图片的接口
    url = 'http://10.59.9.32/jeewms/randCodeImage'

    # 获取验证码图片 stream=True:下载接口的响应
    response = requests.get(url, stream=True)
    # 提取JSESSIONID
    jseesion_id = response.cookies.get("JSESSIONID")

    img = Image.open(BytesIO(response.content))

    # 识别
    code = pytesseract.image_to_string(img, config='--psm 7 -c tessedit_char_whitelist=0123456789').strip()

    return code, jseesion_id